from fastapi import APIRouter

from core.model_core import use_model, generate_xhs_by_subject
from entity.schema.model_schema import ModelInput, GenerateXhs
from utils.base_response import BaseResponse

model=APIRouter()

@model.post("/chat")
async def chat(message:ModelInput):
    """
    使用 DeepSeek V3 自动生成文本
    :param input: 用户输入
    :param session_id: 会话id
    :return: 生成的文本内容
    """
    model_content=await use_model(message.user_input,message.session_id)
    return BaseResponse(code=200,message="成功",data=model_content)


@model.post("/generate/xhs")
async def generate_xhs(message:GenerateXhs):
    """
     使用 DeepSeek V3 自动生成文本
     :param user_input: 用户输入文本
     :return: 生成的文本内容
    """
    request=await generate_xhs_by_subject(message.user_input)
    return BaseResponse(code=200,message="成功",data=request)